import requests

def get_embedding(text: str) -> list:
    """使用 Ollama 的 HTTP API 获取文本嵌入"""
    response = requests.post(
        'http://localhost:11434/api/embeddings',
        json={
            'model': 'bge-m3',
            'prompt': text
        }
    )
    return response.json()['embedding']

# 示例使用
text = "这是一个示例文本"
embedding = get_embedding(text)
print(embedding)
